D:\a\tools.proto\tools.proto\compiler\src\compiler\util\types.rs
Line | Count | Source |
1 | | // Copyright (c) 2024, BlockProject 3D |
2 | | // |
3 | | // All rights reserved. |
4 | | // |
5 | | // Redistribution and use in source and binary forms, with or without modification, |
6 | | // are permitted provided that the following conditions are met: |
7 | | // |
8 | | // * Redistributions of source code must retain the above copyright notice, |
9 | | // this list of conditions and the following disclaimer. |
10 | | // * Redistributions in binary form must reproduce the above copyright notice, |
11 | | // this list of conditions and the following disclaimer in the documentation |
12 | | // and/or other materials provided with the distribution. |
13 | | // * Neither the name of BlockProject 3D nor the names of its contributors |
14 | | // may be used to endorse or promote products derived from this software |
15 | | // without specific prior written permission. |
16 | | // |
17 | | // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
18 | | // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
19 | | // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
20 | | // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
21 | | // CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
22 | | // EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
23 | | // PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
24 | | // PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
25 | | // LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
26 | | // NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
27 | | // SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
28 | | |
29 | | use std::borrow::Cow; |
30 | | use std::collections::HashMap; |
31 | | use std::ops::Deref; |
32 | | use std::rc::Rc; |
33 | | |
34 | | pub trait TypeMapper { |
35 | | fn map_local_type<'a>(&self, item_type: &'a str) -> Cow<'a, str>; |
36 | | fn map_foreign_type<'a>(&self, item_type: &'a str) -> Cow<'a, str>; |
37 | | } |
38 | | |
39 | | pub trait Name { |
40 | | fn name(&self) -> &str; |
41 | | } |
42 | | |
43 | | pub trait PtrKey { |
44 | | fn ptr_key(&self) -> usize; |
45 | | } |
46 | | |
47 | | impl<K> PtrKey for Rc<K> { |
48 | 447 | fn ptr_key(&self) -> usize { |
49 | 447 | &**self as *const K as usize |
50 | 447 | } |
51 | | } |
52 | | |
53 | | impl<K: Name> Name for Rc<K> { |
54 | 488 | fn name(&self) -> &str { |
55 | 488 | self.deref().name() |
56 | 488 | } |
57 | | } |
58 | | |
59 | | #[derive(Clone, Debug)] |
60 | | pub struct TypePathMap { |
61 | | type_path_by_addr: HashMap<usize, String>, |
62 | | } |
63 | | |
64 | | impl Default for TypePathMap { |
65 | 0 | fn default() -> Self { |
66 | 0 | Self::new() |
67 | 0 | } |
68 | | } |
69 | | |
70 | | impl TypePathMap { |
71 | 55 | pub fn new() -> Self { |
72 | 55 | Self { |
73 | 55 | type_path_by_addr: HashMap::new(), |
74 | 55 | } |
75 | 55 | } |
76 | | |
77 | 38 | pub fn add<K: PtrKey>(&mut self, key: &K, type_path: String) { |
78 | 38 | self.type_path_by_addr.insert(key.ptr_key(), type_path); |
79 | 38 | } |
80 | | |
81 | 357 | pub fn get<'a, K: PtrKey + Name, T: TypeMapper>(&'a self, mapper: &T, item_type: &'a K) -> Cow<'a, str> { |
82 | 357 | match self.type_path_by_addr.get(&item_type.ptr_key()) { |
83 | 190 | None => mapper.map_local_type(item_type.name()), |
84 | 167 | Some(v) => mapper.map_foreign_type(v), |
85 | | } |
86 | 357 | } |
87 | | } |